# import pandas as pd
# df = pd.read_excel('产品销售表.xlsx', sheet_name=['第1分店','第2分店','第3分店'])
# df=pd.concat([df['第1分店'],df['第2分店'],df['第3分店']])
# df.dropna(axis=0,subset=['数量'],inplace=True)
# df.drop_duplicates(inplace=True)
# print(df.groupby('产品名称').agg('sum')[['数量','销售额（万元）']])
# print(df.groupby('分店名称').agg('sum')[['数量','销售额（万元）']])
# print(df.groupby('季度').agg('sum')[['数量', '销售额（万元）']])
import pandas as pd
import os  # 导入os模块

# 数据处理部分（保持不变）
df = pd.read_excel('第4分店产品销售表.xlsx')
df1 = pd.DataFrame()
groups = df.groupby('单价（元）')
for group in groups:
    df_temp = pd.DataFrame(group[1])
    df_temp.fillna(method='bfill', inplace=True)
    df1 = pd.concat([df1, df_temp])
df1.drop_duplicates(inplace=True)

df2 = pd.DataFrame()
groups = df1.groupby('季度')
for group in groups:
    df_temp = pd.DataFrame(group[1])
    df2 = pd.concat([df2, df_temp], ignore_index=True)
df2.insert(0, '分店名称', '第4分店')

# 写入Excel（修改路径判断方式）
# 使用os.path.exists判断文件是否存在
mode = 'a' if os.path.exists('产品销售表.xlsx') else 'w'

writer = pd.ExcelWriter(
    '产品销售表.xlsx',
    engine='openpyxl',
    mode=mode,
    if_sheet_exists='replace'
)

df2.to_excel(writer, sheet_name='第4分店', index=False)

writer.close()